# -*- coding: utf-8 -*-
# @Time: 2024/10/27 13:38
# @Author: foxhuty
# @File: files_test.py
import pandas as pd

df = pd.DataFrame({'name': ['tommy', 'huty', 'may', 'jenipher'], 'sex': ['male', 'male', 'female', 'female'],
                   'age': [18, 18, 19, 19]})
# print(df)
# print(df.columns)
#
# # for item in df.columns:
# #     new_col=item+'占比'
# #     print(new_col)
# df.columns.tolist()
# print(df.columns.tolist())
# new_col = [col + '_percent' for col in df.columns]
# print(new_col)
# col_dict = dict(zip(df.columns, new_col))
# # col_zip = zip(df.columns.tolist(), new_col)
# print(col_dict)
# df.rename(columns=col_dict,inplace=True)
# print(df.columns)
# new_list=[]
columns = ['参考人数', '语文', '语文占比', '数学', '数学占比', '英语', '英语占比', '物理', '物理占比', '历史',
           '历史占比',
           '政治', '政治占比', '地理', '地理占比', '生物', '生物占比', '化学', '化学占比', '总分', '总分占比']
col_history = ['总分', '语文', '数学', '英语', '历史', '生物', '政治', '地理',
               '参考人数', '总分占比', '语文占比', '数学占比', '英语占比', '历史占比', '生物占比', '政治占比',
               '地理占比']
col_his_order = ['参考人数', '语文', '语文占比', '数学', '数学占比', '英语', '英语占比', '历史', '历史占比', '生物',
                 '生物占比', '政治', '政治占比', '地理', '地理占比', '总分', '总分占比']

col_phys_order = ['参考人数', '语文', '语文占比', '数学', '数学占比', '英语', '英语占比', '历史', '历史占比', '生物',
                  '生物占比', '化学', '化学占比', '政治', '政治占比', '地理', '地理占比', '总分', '总分占比']
col_physics = ['总分', '语文', '数学', '英语', '物理', '化学', '生物', '政治', '地理',
               '参考人数', '总分占比', '语文占比', '数学占比', '英语占比', '物理占比', '化学占比', '生物占比',
               '政治占比', '地理占比']
print(col_physics[col_physics.index('参考人数')])
print(col_history.index('参考人数'))


def custom_sort_key(word, order_list):
    if word in order_list:
        return order_list.index(word)
    else:
        return len(order_list)


col_history.sort(key=lambda x: custom_sort_key(x, col_his_order))
col_physics.sort(key=lambda x: custom_sort_key(x, col_phys_order))
print(col_history)
print(col_physics)
